Swfit-05.Array的常见操作

iOS开发
Swift

  • map / filter / reduce

    1
    2
    3
    4
    5
    6
    7
    8
    var arr = [1, 2, 3, 4]
    // [2, 4, 6, 8]
    var arr2 = arr.map { $0 * 2 }
    // [2, 4]
    var arr3 = arr.filter { $0 % 2 == 0 } // 10
    var arr4 = arr.reduce(0) { $0 + $1 } // 10
    // 可简写为
    var arr5 = arr.reduce(0, +)

  • map接收函数作为参数

    1
    2
    3
    4
    func double(_ i: Int) -> Int { i * 2 } 
    var arr = [1, 2, 3, 4]
    // [2, 4, 6, 8] 
    print(arr.map(double))

  • compactMap默认解包并去除nil

    1
    2
    3
    4
    5
    6
    7
    var arr = ["123", "test", "jack", "-30"] 

    // [Optional(123), nil, nil, Optional(-30)] 
    var arr2 = arr.map { Int($0) }

    // [123, -30]
    var arr3 = arr.compactMap { Int($0) }

  • 使用reduce实现map、filter函数的功能

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    var arr = [1, 2, 3, 4]

    // [2, 4, 6, 8]
    print(arr.map { $0 * 2 }) 
    // 等价于👆函数功能
    print(arr.reduce([]) { $0 + [$1 * 2] })

    // [2, 4]
    print(arr.filter { $0 % 2 == 0 }) 
    // 等价于👆函数功能
    print(arr.reduce([]) { $1 % 2 == 0 ? $0 + [$1] : $0 })

  • map,flatMap功能对比

    1
    2
    3
    4
    5
    var arr = [1, 2, 3]
    // [[1], [2, 2], [3, 3, 3]] 拿到什么(数组)，返回数组(二维数组)就放什么，不做处理。
    var arr2 = arr.map { Array.init(repeating: $0, count: $0) }
    // [1, 2, 2, 3, 3, 3] 将拿到的数组中的元素取出放到返回数组(一维数组)中。扁平化处理。
    var arr3 = arr.flatMap { Array.init(repeating: $0, count: $0) }

  • 用lazy的优化map

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    let arr = [1, 2, 3]
    let result = arr.lazy.map {
    	(i: Int) -> Int in 
    	print("mapping \(i)") 
    	return i * 2
    }
    print("begin-----")
    print("mapped", result[0]) 
    print("mapped", result[1]) 
    print("mapped", result[2]) 
    print("end----")
    // 第一次用到时，才会执行相应逻辑
    begin----- 
    mapping 1 
    mapped 2 
    mapping 2 
    mapped 4 
    mapping 3 
    mapped 6 
    end----

  • map和flatMap可选值Optional处理中的应用

    1
    2
    3
    4
    5
    6
    7
    var num1: Int? = 10
    // Optional(20)
    var num2 = num1.map { $0 * 2 }

    var num3: Int? = nil
    // nil，发现$0为nil直接返回nil
    var num4 = num3.map { $0 * 2 }

map 和 flatMap对比

1
2
3
4
5
6
var num1: Int? = 10
// Optional(Optional(20)), map还是会对可选值再包装一层
var num2 = num1.map { Optional.some($0 * 2) }

// Optional(20), flatMap发现已经是可选值，就不会再包装了
var num3 = num1.flatMap { Optional.some($0 * 2) }

可用于简化代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
// 示例一：
var num1: Int? = 10
var num2 = (num1 != nil) ? (num1! + 10) : nil 
var num3 = num1.map { $0 + 10 }
// num2、num3是等价的

// 示例二：
var fmt = DateFormatter()
fmt.dateFormat = "yyyy-MM-dd"
var str: String? = "2011-09-10"
// old
var date1 = str != nil ? fmt.date(from: str!) : nil 
// new
var date2 = str.flatMap(fmt.date)

// 示例三：
var score: Int? = 98
// old
var str1 = score != nil ? "socre is \(score!)" : "No score" 
// new
var str2 = score.map { "score is \($0)" } ?? "No score"

// 示例四：
struct Person {
    var name: String
    var age: Int
}
var items = [
	Person(name: "jack", age: 20), 
	Person(name: "rose", age: 21), 
	Person(name: "kate", age: 22)
]
// old
func getPerson1(_ name: String) -> Person? {
	let index = items.firstIndex { $0.name == name } 
	return index != nil ? items[index!] : nil
}
// new
func getPerson2(_ name: String) -> Person? {
	return items.firstIndex { $0.name == name }.map { items[$0] }
}

// 示例五：
struct Person {
	var name: String
	var age: Int
	init?(_ json: [String : Any]) {
		guard let name = json["name"] as? String, 
		let age = json["age"] as? Int else {
			return nil
		}
		self.name = name 
		self.age = age
	} 
}
var json: Dictionary? = ["name" : "Jack", "age" : 10] 
// old
var p1 = json != nil ? Person(json!) : nil
// new
var p2 = json.flatMap(Person.init)

Share